#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@file    __init__.py
@brief
@details
@author  Shivelino
@date    2023-12-23 19:06
@version 0.0.1

@par Copyright(c):
@par todo:
@par history:
"""
from .lenet import LeNet
from .mobilenet import MobileNet


def get_model(model_name: str):
    model_name = model_name.lower()
    if model_name == "lenet":
        return LeNet()
    elif model_name == "mobilenet":
        return MobileNet()
    else:
        raise Exception("unsupported model")


__version__ = "0.0.1"
__all__ = [
    "get_model",
    "LeNet",
    "MobileNet"
]
__author__ = "Shivelino <shivelino@qq.com>"

import sys
import os.path as op

sys.path.append(op.dirname(__file__))
